String: [
  // Current methods are good, but could add:
  'internet.password',        // For password fields
  'internet.displayName',     // For display names/usernames
  'internet.emoji',           // For emoji/reaction fields
  'system.fileName',          // For file name fields
  'system.filePath',          // For file path fields
  'system.mimeType',          // For content type fields
  'book.title',               // For title fields
  'music.songName',           // For song/media titles
  'food.dish',                // For food-related strings
  'animal.type',              // For animal/pet fields
  'vehicle.model',            // For vehicle/product models
  'hacker.phrase',            // For tech-related descriptions
  'science.chemicalElement',  // For scientific data
]

Number: [
  // Current methods are good, but could add:
  'number.binary',            // For binary numbers
  'number.octal',             // For octal numbers
  'number.hex',               // For hexadecimal numbers
  'commerce.price',           // For price fields (returns number)
  'date.weekday',             // For day numbers
  'internet.port',            // For port numbers
]

// Add this state near the other useState declarations (around line 61)
const [originalLlmData, setOriginalLlmData] = useState<Record<string, {
  mongoType: MongoDBFieldType;
  fakerMethod: string;
  fakerArgs: any[];
}>>({});

// Add this useEffect after the existing state declarations
useEffect(() => {
  const originalData = Object.fromEntries(
    Object.entries(fakerSchema).map(([field, mapping]) => [
      field,
      {
        mongoType: mapping.mongoType,
        fakerMethod: mapping.fakerMethod,
        fakerArgs: mapping.fakerArgs
      }
    ])
  );
  setOriginalLlmData(originalData);
}, [fakerSchema]);

// Update the onJsonTypeSelect function (around line 73)
const onJsonTypeSelect = (newJsonType: MongoDBFieldType) => {
  const currentMapping = fakerSchemaFormValues[activeField];
  const originalData = originalLlmData[activeField];
  
  if (currentMapping) {
    // If switching back to original type, restore original LLM data
    const shouldRestoreOriginal = originalData && 
      newJsonType === originalData.mongoType &&
      originalData.fakerMethod !== UNRECOGNIZED_FAKER_METHOD;
    
    const fakerMethod = shouldRestoreOriginal 
      ? originalData.fakerMethod 
      : getDefaultFakerMethod(newJsonType);
      
    const fakerArgs = shouldRestoreOriginal 
      ? originalData.fakerArgs 
      : [];

    setFakerSchemaFormValues({
      ...fakerSchemaFormValues,
      [activeField]: {
        ...currentMapping,
        mongoType: newJsonType,
        fakerMethod,
        fakerArgs,
      },
    });
    resetIsSchemaConfirmed();
  }
};

// Update the onFakerFunctionSelect function (around line 91)
const onFakerFunctionSelect = (newFakerFunction: string) => {
  const currentMapping = fakerSchemaFormValues[activeField];
  const originalData = originalLlmData[activeField];
  
  if (currentMapping) {
    // Restore original args if user reselects original LLM method
    const fakerArgs = (originalData?.fakerMethod === newFakerFunction) 
      ? originalData.fakerArgs 
      : currentMapping.fakerArgs;

    setFakerSchemaFormValues({
      ...fakerSchemaFormValues,
      [activeField]: {
        ...currentMapping,
        fakerMethod: newFakerFunction,
        fakerArgs,
      },
    });
    resetIsSchemaConfirmed();
  }
};